package com.alexd.leetcode.contest.week102;

/*904. 水果成篮
用户通过次数 5
用户尝试次数 10
通过次数 5
提交次数 13
题目难度 Medium
在一排树中，第 i 棵树产生 tree[i] 型的水果。
你可以从你选择的任何树开始，然后重复执行以下步骤：

把这棵树上的水果放进你的篮子里。如果你做不到，就停下来。
移动到当前树右侧的下一棵树。如果右边没有树，就停下来。
请注意，在选择一颗树后，你没有任何选择：你必须执行步骤 1，然后执行步骤 2，然后返回步骤 1，然后执行步骤 2，依此类推，直至停止。

你有两个篮子，每个篮子可以携带任何数量的水果，但你希望每个篮子只携带一种类型的水果。
用这个程序你能收集的水果总量是多少？

 

示例 1：

输入：[1,2,1]
输出：3
解释：我们可以收集 [1,2,1]。
示例 2：

输入：[0,1,2,2]
输出：3
解释：我们可以收集 [1,2,2].
如果我们从第一棵树开始，我们将只能收集到 [0, 1]。
示例 3：

输入：[1,2,3,2,2]
输出：4
解释：我们可以收集 [2,3,2,2].
如果我们从第一棵树开始，我们将只能收集到 [1, 2]。
示例 4：

输入：[3,3,3,1,2,1,1,2,3,3,4]
输出：5
解释：我们可以收集 [1,2,1,1,2].
如果我们从第一棵树或第八棵树开始，我们将只能收集到 4 个水果。
 

提示：

1 <= tree.length <= 40000
0 <= tree[i] < tree.length
*/
public class Solution2 {
	public int totalFruit(int[] tree) {
		int one = tree[0],two = -1;
		int max = -1;
		int cnt = 0;
		int firstTwoIdx = -1;
        for (int i = 0; i < tree.length;) {
			if(tree[i] == one || tree[i] == two) {
				/*if(tree[i] == two) {
					lastTwoIdx = i;
				}*/
				i++;
				cnt++;
				
			}else if(tree[i] != one && tree[i] != two){
				if(two == -1) {
					two = tree[i];
					firstTwoIdx = i;
					i++;
					cnt++;
				}else {
					max = Math.max(max, cnt);
					i = firstTwoIdx;
					cnt = 0;
					one = tree[i];
					two = -1;
				}
			}
		}
        max = Math.max(max, cnt);
        return max;
    }
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		int[] data1 = {3,3,3,1,2,1,1,2,3,3,4};
		if(5 == new Solution2().totalFruit(data1)) {
			System.out.println("testcase1 KO");
		}
		int[] data2 = {1,2,3,2,2};
		if(4 == new Solution2().totalFruit(data2)) {
			System.out.println("testcase2 KO");
		}
		int[] data3 = {0,1,2,2};
		if(3 == new Solution2().totalFruit(data3)) {
			System.out.println("testcase3 KO");
		}
		int[] data4 = {1,2,1};
		if(3 == new Solution2().totalFruit(data4)) {
			System.out.println("testcase4 KO");
		}

		int[] data5 = {0,1,6,6,4,4,6};
		if(5 == new Solution2().totalFruit(data5)) {
			System.out.println("testcase5 KO");
		}
				
	}

}
